from fastapi.testclient import TestClient
from main import app

client = TestClient(app)


def test_chat():
    response = client.get("/ai_wenxin/chat", params={"content": "介绍一下北京"})
    assert response.status_code == 200
    assert "response" in response.json()
